package com.zhoukx.netty.grantChat;

import io.netty.channel.Channel;

import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.text.SimpleDateFormat;
import java.util.*;


public class GroupChatServerHandler extends SimpleChannelInboundHandler<String> {

     public static List<Channel> channels = new ArrayList<>();

     // 使用一个hashMap管理
    //public static Map<String,Channel> channels = new HashMap<String,Channel>();

    // 定义一个channel组,管理所有的channel
    // GlobalEventExecutor.INSTANCE 是全局事件执行器,是一个单例
    private  static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");


    /**
     * handlerAdded 表示连接建立,一旦连接,第一个被执行
     * 将当前channel加入到channelGroup
     * @throws Exception
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx)  throws Exception {

        Channel channel = ctx.channel();

        // 将该客户加入聊天信息推送给其他的在线客户
        /**
         * 该方法会将channelGroup中所有的channel遍历,并发送信息
         *  不需要自己遍历
         */
        channelGroup.writeAndFlush("[客户端]"+channel.remoteAddress()+" 加入聊天" + sdf.format(new Date())+"\n");

        channelGroup.add(channel);
    }

    /**
     * 断开连接,将客户离开的信息推送给当前的客户
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        channelGroup.writeAndFlush("[客户端]"+channel.remoteAddress()+" 离开了");
        System.out.println("channelGroup size"+channelGroup.size());
    }

    /**
     *  表示channel处于不活动状态,提示xx上线
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress()+" 上线");
    }
    /**
     *  表示channel处于不活动状态,提示xx离线
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress()+" 离线了");
    }


    /**
     *  读取数据方法
     * @param ctx
     * @param msg
     * @throws Exception
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {

        // 获取当前channel
        Channel channel = ctx.channel();

        // 这时我们遍历channelGroup,根据情况的不同,回送不同的消息

        channelGroup.forEach(ch ->{
            if (channel != ch) {  // 不是当前的channel,转发消息
                ch.writeAndFlush("[客户]"+channel.remoteAddress()+ " 发送了消息:"+msg+"\n");
            }else { //回显自己发送的消息给自己
                ch.writeAndFlush("[自己]发送了消息:"+msg+"\n");
            }
        });
    }


    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        ctx.close();
    }
}
